<?php

namespace summer\swoole\pool\utils;

use Swoole\Coroutine;
use Swoole\Coroutine as SwooleCoroutine;
use Closure;
use ArrayObject;

class ContextUtil
{
    protected static $nonCoContext = [];

    public static function set(string $id, $value)
    {
        if (Coroutine::getCid() > 0) {
            self::getContextFor()[$id] = $value;
        } else {
            static::$nonCoContext[$id] = $value;
        }
        return $value;
    }

    public static function get(string $id, $default = null, $coroutineId = null)
    {
        if (Coroutine::getCid() > 0) {
            return self::getContextFor($coroutineId)[$id] ?? $default;
        }

        return static::$nonCoContext[$id] ?? $default;
    }

    public static function has(string $id, $coroutineId = null)
    {
        if (Coroutine::getCid() > 0) {
            return isset(self::getContextFor($coroutineId)[$id]);
        }

        return isset(static::$nonCoContext[$id]);
    }

    /**
     * Release the context when you are not in coroutine environment.
     */
    public static function destroy(string $id)
    {
        unset(static::$nonCoContext[$id]);
    }

    /**
     * Copy the context from a coroutine to current coroutine.
     */
    public static function copy(int $fromCoroutineId, array $keys = []): void
    {
        $from = self::getContextFor($fromCoroutineId);
        if ($from === null) {
            return;
        }

        $current = self::getContextFor();

        $tempKeys = array_intersect_key($from->getArrayCopy(), array_flip((array)$keys));

        $current->exchangeArray($keys ? $tempKeys : $from->getArrayCopy());
    }

    /**
     * Retrieve the value and override it by closure.
     */
    public static function override(string $id, Closure $closure)
    {
        $value = null;
        if (self::has($id)) {
            $value = self::get($id);
        }
        $value = $closure($value);
        self::set($id, $value);
        return $value;
    }

    /**
     * Retrieve the value and store it if not exists.
     * @param mixed $value
     */
    public static function getOrSet(string $id, $value)
    {
        if (!self::has($id)) {
            $theValue = $value instanceof Closure ? $value() : $value;
            return self::set($id, $theValue);
        }
        return self::get($id);
    }

    public static function getContainer()
    {
        if (Coroutine::getCid() > 0) {
            return self::getContextFor();
        }

        return static::$nonCoContext;
    }

    /**
     * @return null|ArrayObject
     */
    public static function getContextFor(?int $id = null)
    {
        if ($id === null) {
            return SwooleCoroutine::getContext();
        }

        return SwooleCoroutine::getContext($id);
    }
}